/* -------------------------------------------------------------------------- *
   TestWinUnit - Maria Blees (maria.blees@microsoft.com)

   File: TestablePlainTextLogger.h

   Implementation of PlainTextLogger that writes to a 1K buffer.
 * -------------------------------------------------------------------------- */
#pragma once

#include "PlainTextLogger.h"

class TestablePlainTextLogger : public WinUnitLib::PlainTextLogger
{
private:
    char _buffer[1024];
public:
    TestablePlainTextLogger();
public:
    virtual PlainTextLogger& operator<<(const wchar_t* str);
    virtual PlainTextLogger& operator<<(const char* str);
    virtual PlainTextLogger& operator<<(unsigned int i);
public:
    void ClearBuffer();
    char* GetBuffer(); // Can't be const because I use strtok_s
};